import java.util.HashSet;
import java.util.Set;

public class HappyNum {
    /*
    * 快乐数
    * 编写一个算法来判断一个数 n 是不是快乐数。
    * 快乐数 定义为：
    * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
    * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
    * 如果这个过程 结果为 1，那么这个数就是快乐数。
    * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
    * */
    public static void main(String[] args){

    }

    // 我的想法：无限循环，说明会重复出现，一旦重复，return false就好
    public boolean mySolution(int n){
        int sum = 0;
        Set<Integer> set = new HashSet();
        while (true){
            String s = n + "";
            for (int i = 0; i < s.length(); i++) {
                int value = Integer.parseInt(String.valueOf(s.charAt(i)));
                sum += value*value;
            }
            if(sum == 1) {
                return true;
            } else {
                if (set.contains(sum))
                    break;
                else {
                    set.add(sum);
                    n = sum;
                    sum = 0;
                }
            }
        }
        return false;
    }

    public boolean solution(int n){
        Set<Integer> set = new HashSet<>();
        while (n != 1 && !set.contains(n)){
            set.add(n);
            n = getNextValue(n);
        }
        return n==1;
    }

    // 别搁哪字符串取值，直接计算快多了
    private int getNextValue(int n){
        int res = 0;
        while (n>0){
            int i = n%10;
            res += i*i;
            n = n/10;
        }
        return res;
    }
}
